# -*- coding: utf-8 -*-
# @Time : 2022/1/6 10:25
# @Author : Limusen
# @File : login

from selenium import webdriver
from common.base_page import BasePage
from element_infos.login.login_page import LoginPage


class Login:

    def __init__(self, driver):
        self.login = LoginPage(driver)

    def login_action(self, name, psw):
        # 直接调用login_page中的方法一
        self.login.login_action(name, psw)
        return self.login.get_page_source()

    def logins(self, name, psw):
        # 自己组装login_page中的方法
        self.login.click_main_login()
        self.login.input_username(name)
        self.login.input_password(psw)
        self.login.click_login()
        return self.login.get_page_source()


if __name__ == '__main__':
    driver = webdriver.Chrome()
    base = BasePage(driver)
    login = Login(driver)
    base.open_url("http://shop.aircheng.com/")
    base.implicitly_wait(10)
    # login.login_action("nswe", "111111") # 方式一

    login.logins("nswe", "111111")  # 方式二
